/*
 * Copyright (c) 2023 VMware, Inc. or its affiliates, All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.tensai.study.framework.reactornetty.udp;

import lombok.extern.slf4j.Slf4j;
import reactor.core.publisher.Sinks;
import reactor.netty.Connection;
import reactor.netty.udp.UdpClient;
import reactor.util.concurrent.Queues;

import java.util.Scanner;

/**
 * @author Tensai
 */
@Slf4j
public class QuoteOfTheMomentClient {

	private static final int PORT = 7686;
	private static final boolean WIRETAP = false;

	static final Sinks.Many<String>
			SINKS = Sinks.many().multicast().onBackpressureBuffer(Queues.SMALL_BUFFER_SIZE, false);

	public static void main(String[] args) {
		UdpClient client =
				UdpClient.create()
				         .port(PORT)
				         .wiretap(WIRETAP);

		Connection conn = client.connectNow();

		conn.inbound()
		    .receive()
		    .asString()
		    .doOnNext(text -> log.info("收到消息：{}", text))
		    .doOnError(err -> {
		        log.error(err.getMessage(), err);
		        conn.disposeNow();
		    })
		    .subscribe();

		conn.outbound()
		    .sendString(SINKS.asFlux())
		    .then()
		    .subscribe();

		conn.onReadIdle(500000, () -> {
			log.error("消息超时");
			conn.disposeNow();
		});

		while (true) {
			Scanner scanner = new Scanner(System.in);
			String input = scanner.nextLine();
			if (!"exit".equals(input)){
				SINKS.tryEmitNext(input);
			} else {
				conn.onDispose()
				    .block();
				return;
			}
		}


	}
}